﻿using System;
using System.Threading;
using System.Threading.Tasks;
using System.Windows;

namespace Util
{
    public class ExecuteUtil
    {
        /// <summary>
        /// 执行一个方法，并且限制其执行时间
        /// </summary>
        /// <param name="action">执行的方法</param>
        /// <param name="timeout">执行时间限制</param>
        /// <returns>是否超时</returns>
        public static bool ExecuteWithTimeout(Action action, TimeSpan timeout)
        {
            var cts = new CancellationTokenSource();
            var token = cts.Token;

            var task = Task.Run(() =>
            {
                action.Invoke();

                if (token.IsCancellationRequested)
                {
                    return;
                }
            });

            if (Task.WaitAny(new[] { task }, timeout) == -1)
            {
                cts.Cancel();
                return false;
            }

            try
            {
                task.Wait();
                return true;
            }
            catch (AggregateException e)
            {
                MessageBox.Show(e.Message, "方法执行错误");
                return false;
            }
        }

        public static async Task<bool> ExecuteWithTimeoutAsync(Func<Task> asyncAction, TimeSpan timeout)
        {
            var timeoutTask = Task.Delay(timeout);
            var actionTask = asyncAction();

            var completedTask = await Task.WhenAny(timeoutTask, actionTask);

            if (completedTask == timeoutTask)
            {
                MessageBox.Show("方法执行超时！");
                return false;
            }

            try
            {
                await actionTask;
                return true;
            }
            catch (Exception e)
            {
                MessageBox.Show(e.Message, "方法执行错误");
                return false;
            }
        }

        public class ResultWrapper
        {
            public ushort[] UshortArrayValue { get; set; }
            public bool BoolValue { get; set; }
            public int IntValue { get; set; }
            public double DoubleValue { get; set; } = -9999;
        }
        /// <summary>
        /// 执行一个方法，并且限制其执行时间,并且返回执行结果
        /// </summary>
        /// <param name="action">执行的方法</param>
        /// <param name="timeout">执行时间限制</param>
        /// <param name="res">执行结果</param>
        /// <returns>是否超时</returns>
        public static bool ExecuteWithTimeout(Func<ushort[]> method, TimeSpan timeout, out ushort[] res)
        {
            var cts = new CancellationTokenSource();
            var token = cts.Token;

            var resultWrapper = new ResultWrapper(); // 创建包装类实例

            var task = Task.Run(() =>
            {
                try
                {
                    resultWrapper.UshortArrayValue = method(); // 执行带有返回值的方法，并将返回值存储在包装类实例中
                }
                catch (Exception ex)
                {
					Helpers.Logger.Instance.Warning($"Catch ExecuteWithTimeout {ex.Message}");
				}


                if (token.IsCancellationRequested)
                {
                    return;
                }
            });

            if (Task.WaitAny(new Task[] { task }, timeout) == -1)
            {
                cts.Cancel();
                res = null;
                return false; // 超时，返回 false
            }

            try
            {
                task.Wait();
                res = resultWrapper.UshortArrayValue; // 将包装类中的返回值赋给 res
                return true; // 任务完成，返回 true
            }
            catch (AggregateException e)
            {
                MessageBox.Show(e.Message, "方法执行错误");
                res = null;
                return false; // 任务抛出异常，返回 false
            }
        }

        /// <summary>
        /// 执行一个方法，并且限制其执行时间,并且返回执行结果
        /// </summary>
        /// <param name="action">执行的方法</param>
        /// <param name="timeout">执行时间限制</param>
        /// <param name="res">执行结果</param>
        /// <returns>是否超时</returns>
        public static bool ExecuteWithTimeout(Func<bool> method, TimeSpan timeout, out bool res)
        {
            var cts = new CancellationTokenSource();
            var token = cts.Token;

            var resultWrapper = new ResultWrapper(); // 创建包装类实例

            var task = Task.Run(() =>
            {
                resultWrapper.BoolValue = method(); // 执行带有返回值的方法，并将返回值存储在包装类实例中

                if (token.IsCancellationRequested)
                {
                    return;
                }
            });

            if (Task.WaitAny(new Task[] { task }, timeout) == -1)
            {
                cts.Cancel();
                res = false;
                return false; // 超时，返回 false
            }

            try
            {
                task.Wait();
                res = resultWrapper.BoolValue; // 将包装类中的返回值赋给 res
                return true; // 任务完成，返回 true
            }
            catch (AggregateException e)
            {
                MessageBox.Show(e.Message, "方法执行错误");
                res = false;
                return false; // 任务抛出异常，返回 false
            }
        }

        /// <summary>
        /// 执行一个方法，并且限制其执行时间,并且返回执行结果
        /// </summary>
        /// <param name="action">执行的方法</param>
        /// <param name="timeout">执行时间限制</param>
        /// <param name="res">执行结果</param>
        /// <returns>是否超时</returns>
        public static bool ExecuteWithTimeout(Func<int, double> method, int parameter, TimeSpan timeout, out double res)
        {
            var cts = new CancellationTokenSource();
            var token = cts.Token;

            var resultWrapper = new ResultWrapper(); // 创建包装类实例

            var task = Task.Run(() =>
            {
                resultWrapper.DoubleValue = method(parameter); // 执行带有参数的方法，并将返回值存储在包装类实例中
                if (token.IsCancellationRequested)
                {
                    return;
                }
            });

            if (Task.WaitAny(new Task[] { task }, timeout) == -1)
            {
                cts.Cancel();
                res = resultWrapper.DoubleValue;
                return false; // 超时，返回 false
            }

            try
            {
                task.Wait();
                res = resultWrapper.DoubleValue; // 将包装类中的返回值赋给 res
                return true; // 任务完成，返回 true
            }
            catch (AggregateException e)
            {
                MessageBox.Show(e.Message, "方法执行错误");
                res = resultWrapper.DoubleValue;
                return false; // 任务抛出异常，返回 false
            }
        }

        /// <summary>
        /// 执行一个方法，并且限制其执行时间,并且返回执行结果
        /// </summary>
        /// <param name="action">执行的方法</param>
        /// <param name="timeout">执行时间限制</param>
        /// <param name="res">执行结果</param>
        /// <returns>是否超时</returns>
        public static bool ExecuteWithTimeout(Func<int> method, TimeSpan timeout, out int res)
        {
            var cts = new CancellationTokenSource();
            var token = cts.Token;

            var resultWrapper = new ResultWrapper(); // 创建包装类实例

            var task = Task.Run(() =>
            {
                resultWrapper.DoubleValue = method(); // 执行带有返回值的方法，并将返回值存储在包装类实例中
                if (token.IsCancellationRequested)
                {
                    return;
                }
            });

            if (Task.WaitAny(new Task[] { task }, timeout) == -1)
            {
                cts.Cancel();
                res = resultWrapper.IntValue;
                return false; // 超时，返回 false
            }

            try
            {
                task.Wait();
                res = resultWrapper.IntValue; // 将包装类中的返回值赋给 res
                return true; // 任务完成，返回 true
            }
            catch (AggregateException e)
            {
                MessageBox.Show(e.Message, "方法执行错误");
                res = resultWrapper.IntValue;
                return false; // 任务抛出异常，返回 false
            }
        }
    }
}
